Added new feature

Hey eliot can you check this out!!