Made some changes for demo purposes. Also changed the look function, but I think I can change it back.