Added walls to game. Slight code cleanup