fixing code after review