Added moving ball code