removed unneeded tests and added one for display