Collapse checkbox directive for Angular.js