Fix feature of simulation recording and plotting