Need to clean up code