revised the stubs to finish assignment