Dropping my code for anyone who wants it!