improved template change directive

1. now works with simple ng-class
2. cleaned related old code from controller