The code can be more simple