Python drugstore simulation