[IMP] remove css and improve code.