Started work on beam search