made small modifications to customize output