Add a close button

pull request
why `@class` and not `#import`? This is not a forward declaration so should be able to use `#import`.