Add energy function and changed time units from GeV to seconds