Optimize the codes..