Fixed seg faults