Finished working code
Added final comments for delivery