UI improvements using AngularJs