Removing leftft over testing code