Added controllers code