add some ifs