Add a directive for resetting the form too