make the code more simple