chore(Unit Tests): working on tests for UI components