Cleaned last commit (removed extra prints to console and imports)