whoops, changing ifs to elifs