Added unit tests. Fixed Linting errors.