Added support for $compile