Cleaned up RW approximation notebook