Python code used in an AI group project