Added code coverage!