Made a code more clear