Update main.py

fixed the calculate function. floats instead of ints.