Turn it into a directive