Finishing up data analysis program; still needs improvement