Issue #3

Fixed issue CodeWork #3