added little nice changes