Added return expr test