Working on implementing if.