Minor fixes
Fix graphics and comments