All functions working. Time to do front-end