Fixed seg-fault, everything works, need to clean-up printing