Implementation of basic physics through pygame