if - else: seems to work; more testing next