added pics and fixing minor bugs