Updated Implementation section