Added two #import statements.

Signed-off-by: wsycharles0o <wsycharles0o@hotmail.com>