Update main.cpp

prawdopodobnie dobrze dzialajaca funkcja liczaca koszt