Beta: add if statements