Adding expected return check output