Move terrain draw function to client UI