working on assignment 3. Set is still not working correctly.