Added controller directive