complete programming assignment 2 - need to test