Switch to a switch case to allow more actions