removed todo add beam search picture