Create unit tests for all components

Coverage 80%