fixed the unit tests. 13 out of 13 now pass