added some walls and abit clean up off the code