added some functionality, I'm super pumped