Segfault bug fixed