Added walls function