Python script to calculate electricity bill