Adds return statement