I'd add curly brackets for better readability (since you've got up to three lines inside after all).