Code works for any N greater than 1.

Visual representation works for N > 1, as long as the output fits in the screen.

Visual representation of how the count_rooms loop runs:

![alt text][example1]

Note how the the corners are processed :)

![alt text][example2]

''' python3 square.py ''' [example1]: https://raw.githubusercontent.com/juozasget/square_challange/master/img/example1.png "Visual example" [example2]: https://raw.githubusercontent.com/juozasget/square_challange/master/img/example2.png "Visual example 2"