Forgot to add final code in writeup. Fixed