For first assignment of the DST course on Coursera