Add import, improve learning