Removed test code and extra lines at the top of the file