Cleanup files and add proper documentation