Second Commit

Added gravity , made code more readable