fixed tests , added type hint