Add very basic timing to the tests

Part of #5 (more work could be done to make these better)