Made the code more beautiful.