Added pragma mark and deleted extra spacing