Use single return statement instead of if/then statements