Almost finished working on the backend