added missed import
added discover_tests