Added floor and wall check to prototype