First maze project - python