add add_penalty