Update command.py

Imports we might need