Solution to Caption Generator design