Some modificaton