More elegant way of creating tests