Implement __set_state() for all AST node classes

Implement __set_state() for all AST node classes #476