Add drawWalls function