Functionality updates for gravity