Added functionality to change gravity