Fix Bug #2,3,4