Would like a draw function if possible