Add code file
Update the readme a little