Made minor changes to finish the product