Maybe we should just leave if statement out? I know it was in the original code, but probably not necessary.