Trying to work return statement