Can we use if statements? This code doesn't look great