Add reset_all functionality