TST: Removing Head from code