Modify angular controller