Added color change to walls as well