V0.1 Minor modifications
Solve the subset 1 problem using sample code with little adaptations.