Working on problem six