lev 1 b: if else now working as expected