         return addMilestoneWith(progress, image: image, shouldLightUp: false)
     }
     
-    override func sizeThatFits(size: CGSize) -> CGSize {
+    override public func sizeThatFits(size: CGSize) -> CGSize {
         let newSize = CGSizeMake(frame.size.width,barHeight);
         return newSize;
     }


import UIKit

class Milestone {
    
    var progress: Float!
    var image: UIImage!

//
//  MilestoneProgressBar.swift
//
//  Created by Virindh Borra on 8/20/16.
//  Copyright © 2016 Virindh Borra. All rights reserved.
