Formating and added a simple test function