I added some functionality

Also potato