edit for styling

better variable names
add additional tests