~working version of friday assignment