added solution for Correctness and the Loop Invariant