Code Review Thougths

Here are some quick thoughts and changes to your repo, I commented on them in my fork couldn't figure out how to do it on your side