Minor Changes after review