minor changes
better output