Added missing unit tests and linted