getting test coverage up to 100%