Fixed output (please check again)