lost .timeState reference and added docstrings