control flow issue - add else