Generate class now complete + updated the link file to use the new
generate class