Improve the code a little bit