moved it into a directive