Added option to customize the code