Modified Arduino code for completion of Sprint 2