Added cout in main :)