Add collision detection, goals and ball restart