Worked a lot on the unit test

Working on issue with the room size one now