@kmfoda I have not found a way to add that penalty yet. As I moved on to other projects soon after I posted the question, I did check if PR is merged as mentioned by @patrickvonplaten .
So it would be great if you continue working on this and post solutions if you find it.