adding more tests and fixing some little bugs
adding more tests over inputs