Added classes connected with controller