Tidy up angular code