started to setup unit tests
structure and comments for better understanding