Updated script with fancy colors, and few more checks