https://topicslearn.com/cs304-assignment-2-solution/