Stock Control Python Assignment