Modify angular code