Adding a directive. Still need to make it dynamic