Add initial bouncing and AI code