#36 filling out unit tests