Adding more menu functionality to test all program assignments or specific assignment