Add header, remove unused code