Add gravity back to rocket