Added more tests. Now the tests tat were described in the task are covered -> Need to add more non-trivial tests.