def main$()$:

print$("$n$)$

# Welcome Message

print$("$Welcome$,$ Please answer the prompts to find out your house cleaning cost."$)$

#Define prices for $4$ different cleaning types

cleaning$\_$prices

'Floors': $40,$

'Windows': $50,$

'Bathrooms': $60,$

'Dusting' : $30$

$\}$

#Define house$/$room size cutoffs

smal$1\_$rooms$\_$cutoff $=3$

medium$\_$rooms$\_$cutoff $=6$

large$\_$rooms$\_$cutoff $=9$

#Define what the price will be based on different house sizes

small$\_$house price $=175$

medium house price $=250$

large house price $=300$

#Prompt the User for the number of rooms and cleaning type

num$\_$rooms $=$ int$($input$("$Enter the number of rooms in the house: $"))$

print$("$Select a cleaning type:"$)$

for idx, cleaning$\_$type in enumerate$($cleaning$\_$prices.keys$(),$ start$=1)$:

print $($f$"\{$idx$\}.\{$cleaning$\_$type$\}")$

cleaning$\_$choice $=$ int $($input$("$Enter the number corresponding to the cleaning type: $"))$

#Calculate cost using if statements based on what the user selected for cleaning type and rooms that were input

if num$\_$rooms small$\_$rooms$\_$cutoff:

house$\_$size$\_$price $-=$ smal$\_$ house$\_$price

elif num$\_$rooms medium$\_$rooms$\_$cutoff:

house$\_$size$\_$price $=$ medium$\_$house$\_$price

else:

house$\_$size$\_$price $=$ large$\_$house$\_$price

selected$\_$cleaning $=$ list$($cleaning$\_$prices.keys$())[$cleaning$\_$choice $-1]$

cleaning$\_$cost $=$ cleaning$\_$prices$[$selected$\_$cleaning$]$

#Calculate total cost

total$\_$cost $=$ house$\_$size$\_$price $+$ cleaning$\_$cost

#show the results

print$($f$"$The cost of house cleaning for $\{$num$\_$rooms$\}$ rooms with $\{$selected$\_$cleaning$\}$ is $$\{$total$\_$cost$\}")$

main$()$

Rewrite the python code adding in a function for the option of light cleaning or heavy cleaning