Add necessary use statements, clean up the run-on return statement that was forming...