Done

Just added some comments