added TODO comment
added more steps of the algorithm