work on if - else; in progress (not correct as is)