Calculate electricity bill using Python