First attempt at solution

prior to testing