Added message for the user after calculations. Also added small things to the code.