improved it :)