Improved the audit UX

We've also made changes to the audit, to make it more UX friendly