fixing the code so it works the way it is suppose too