Cleanup code a bit and added a description to the README file