Simplified the return statement