if/elif fix