Updated the implementation and methodology sections