Add a simple main testing, code coverage: 87%
Use more proper assert methods
Add main function in tests