Updated small prototype