Added class to gravity ball