Add cap test
Finish getting cap working