Used FormStateInterface class for  argument type.