Added function to set up walls