Finalized functionality in homework 2