Added return statement to #6