In order to combat the detrimental effect of augmentation on top-5 and top-10 results the beam width could be increased significantly. This would essentially counter the reduction in molecular diversity by sampling more molecules. However, the amount of computation required scales linearly with the beam width; increasing the beam width from 10 to 50 would require five times as much computational resource. Alternatively, Levenshtein augmentation could be used to ensure the input and generated SMILES sequences are similar. This would reduce the likelihood of many SMILES forms being generated -specifically those which are dissimilar to the input -therefore improving molecular diversity.