Add diagonal movement