Edit: Remove testing code