fixed test.py
added feature 1
added feature 2
added more tests