Added import and removed cout

Added import in system.h and removed cout from main.cpp