Implemented nicer graphics