Working! Just need to use rand to generate test data.