Fix fn name
Add diagonal movement