Code changes to provide data attribute to track if design needs to be generated

Code review : Daniel Jao