First assignment for Johns Hopkins Data Science