Clean code, add documentation.