:octocat: Cleaned a bit custom directive