Added walls to joe's code