slight modifications for output