Simple gravity demo with pygame