optimized it a little bit