updates to A functionality of lab3