Add forgoten #imports

XCODE Y U NO COMPLAIN?