Wrote the first steps in the code, need some more work