Small logic error fix in TestingMachine and added new entities to main.