updated the model with helper function to get the completed and the correct output

initial commit