Optimized the script a little bit...