Ui ux

Let's write a directive around this