Refactor the code and added the wall types related images and according to that added the CSS