Remove unnecesarry code