minor changes

just added more tests - no significant changes to functionality
100% coverage guys