add epochs CLI parameter