In my simple pong game I wrote a few years ago, I implemented the AI similar to what you had before, where the computer paddle moves along with ball. Except instead, the computer paddle only started moving when the ball reached the halfway mark across the screen, as any other movement was basically wasted. Calculating the computer paddle's position from the ball data is probably better though.