Update main.c

SOoooooo eine Kombi aus beiden :D
Mit elseif ist das denke ich mal besser.