My second assignment for Biostat 578