Modernize calculate_sigmas function
* Add kl_optimal scheduler * Rename kl_optimal_schedule to kl_optimal_scheduler to be more consistent