minor code edit
proper output!