Remove unneccesarry code