All code functionality working