answer to question part 2 of data_science_toolkit project