added more test cases
mods
assertions passes now