Finished the first assignment