Made the output correct