I was just giving an idea on how this could be done, not on the absolute correctness of the code though.