Added a gravity component so it's not hard-coded