Added code coverage checks and increased coverage beyond 80%