Quick script to simulate homework assignment