Made the code more clearer