Changed main.py to allow eta = 0