added unit case for if..then