cleaned up wall follow code - added states