Fix paddle and ball code