adding some unit tests and updating controller and stylesheet