Update naive_decomposition.py