Update Baby_Care_Media.py

rearrange code