Millorat una mica el codi