Fixes #5.
Change to UIViewController