modified method to pass test..needs work