Add some necessary tests. Make refactoring of tests(forget to make commit before this).