- Small addition to optimizer