Code refactoring
added tests
updated, added pytest
added info about test running