Add option to choose chain length