Simple conditional program, could use some TDD