Changed the code after review simon