Updated adding if case