Changed if for elif