changed input of library and fixed unit test