Bug fix && Makefile update