Seg fault fixed