add epochs option