Refactorized code.
Added walls, preventing ants to move.