Add if-else to next up section