adding more functionality yayy