Register generator enhanced.
Merge pull request #206 from jasvir99/dirty

Register generator enhanced.