Add diamond generation