added coverage and working on unit tests