My submission for the PML course project