changed matplotlib backend, added color option