Proofreading script a bit more