Added some imports to fix compilation