Fix some bugs

Use UIView instead of UIViewController