Cleaned up some imports not used in this script