Optimized the code a bit