Added graphics functionality, currently broken