Added basic gravity implementation