Fix bug & Optimite Code