Seems like you already have this on your to do list but it would be nice to save state on local storage or something like that so that it doesn't reset every time the connection is lost.