Day 5 - updated add button (program.); cleaned up code a bit