added the inital condition from the assignment, checking that my code works properally