Add ability to customize (to add conditions for example)