Adding import (kind of dirty, should find better way).