Adds return; to else statements