* feat: added Unit switch widget with a New unit button * refactor: refactoring after review * refactor: changed the variable name