Make diagram more responsive