Streamlined code, added a gradient thing to the comments