Adapted to cocoapods / Change Implementation to UIView