put some little touches on the code