Remove global state atom & clean up docs