Updated all files with new features