I improve the DataGo class. Also I improve the get_atari_moves method to include remove moves. Also I modify blocks to include the atari state.