Code feedback. Rewrote a few lines to be more pythonic and broke up functions into more independent tests.