refactoring the code

not much just some basics