Organized the code

Moved all the code into `src/`