Add more body functions and move entities in step! function