Added test_optimization.py