Fix bug - add from datetime import datetime