improved testing.py added solution to #5 changed 4