Code seems to be working. Will run overnight and compare plots to be sure.