Added graphics to main and more groundwork