Switched an if to an elif