added some final retouches