Removing innecesary code