Moved imports to inside the class to improve efficiency